/**
 * Unit tests for ChatHandler
 */
import { describe, it, expect, vi, beforeEach } from 'vitest';
import { ChatHandler } from '../../../../src/triggers/handlers/chat-handler';
import { N8nApiClient } from '../../../../src/services/n8n-api-client';
import { InstanceContext } from '../../../../src/types/instance-context';
import { Workflow } from '../../../../src/types/n8n-api';
import { DetectedTrigger } from '../../../../src/triggers/types';
import axios from 'axios';

// Mock getN8nApiConfig
vi.mock('../../../../src/config/n8n-api', () => ({
  getN8nApiConfig: vi.fn(() => ({
    baseUrl: 'https://test.n8n.com/api/v1',
    apiKey: 'test-api-key',
  })),
}));

// Mock SSRFProtection
vi.mock('../../../../src/utils/ssrf-protection', () => ({
  SSRFProtection: {
    validateWebhookUrl: vi.fn(async () => ({ valid: true, reason: '' })),
  },
}));

// Mock buildTriggerUrl
vi.mock('../../../../src/triggers/trigger-detector', () => ({
  buildTriggerUrl: vi.fn((baseUrl: string, trigger: any, mode: string) => {
    return `${baseUrl}/webhook/${trigger.webhookPath}`;
  }),
}));

// Mock axios
vi.mock('axios');

// Create mock client
const createMockClient = (): N8nApiClient => ({
  getWorkflow: vi.fn(),
  listWorkflows: vi.fn(),
  createWorkflow: vi.fn(),
  updateWorkflow: vi.fn(),
  deleteWorkflow: vi.fn(),
  triggerWebhook: vi.fn(),
  getExecution: vi.fn(),
  listExecutions: vi.fn(),
  deleteExecution: vi.fn(),
} as unknown as N8nApiClient);

// Create test workflow
const createWorkflow = (): Workflow => ({
  id: 'workflow-123',
  name: 'Chat Workflow',
  active: true,
  nodes: [
    {
      id: 'chat-node',
      name: 'Chat',
      type: '@n8n/n8n-nodes-langchain.chatTrigger',
      typeVersion: 1,
      position: [0, 0],
      parameters: {
        path: 'ai-chat',
      },
    },
  ],
  connections: {},
  createdAt: new Date().toISOString(),
  updatedAt: new Date().toISOString(),
  settings: {},
  staticData: undefined,
} as Workflow);

describe('ChatHandler', () => {
  let mockClient: N8nApiClient;
  let handler: ChatHandler;

  beforeEach(async () => {
    mockClient = createMockClient();
    handler = new ChatHandler(mockClient);
    vi.clearAllMocks();

    // Reset SSRFProtection mock
    const { SSRFProtection } = await import('../../../../src/utils/ssrf-protection');
    vi.mocked(SSRFProtection.validateWebhookUrl).mockResolvedValue({
      valid: true,
      reason: '',
    });

    // Reset axios mock
    vi.mocked(axios.request).mockResolvedValue({
      status: 200,
      statusText: 'OK',
      data: { response: 'Chat response' },
    });
  });

  describe('initialization', () => {
    it('should have correct trigger type', () => {
      expect(handler.triggerType).toBe('chat');
    });

    it('should have correct capabilities', () => {
      expect(handler.capabilities.requiresActiveWorkflow).toBe(true);
      expect(handler.capabilities.canPassInputData).toBe(true);
    });
  });

  describe('input validation', () => {
    it('should validate correct chat input', () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
        sessionId: 'session-123',
      };

      const result = handler.validate(input);
      expect(result).toEqual(input);
    });

    it('should validate minimal input without sessionId', () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
      };

      const result = handler.validate(input);
      expect(result.workflowId).toBe('workflow-123');
      expect(result.message).toBe('Hello AI!');
      expect(result.sessionId).toBeUndefined();
    });

    it('should reject invalid trigger type', () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'webhook',
        message: 'Hello',
      };

      expect(() => handler.validate(input)).toThrow();
    });

    it('should reject missing message', () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat',
      };

      expect(() => handler.validate(input)).toThrow();
    });

    it('should accept optional fields', () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        data: { context: 'value' },
        headers: { 'X-Custom': 'header' },
        timeout: 60000,
        waitForResponse: false,
      };

      const result = handler.validate(input);
      expect(result.data).toEqual({ context: 'value' });
      expect(result.headers).toEqual({ 'X-Custom': 'header' });
      expect(result.timeout).toBe(60000);
      expect(result.waitForResponse).toBe(false);
    });
  });

  describe('execute', () => {
    it('should execute chat with provided sessionId', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
        sessionId: 'custom-session',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(true);
      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          method: 'POST',
          data: expect.objectContaining({
            action: 'sendMessage',
            sessionId: 'custom-session',
            chatInput: 'Hello AI!',
          }),
        })
      );
    });

    it('should generate sessionId when not provided', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(true);
      expect(response.metadata?.sessionId).toMatch(/^session_\d+_[a-z0-9]+$/);
    });

    it('should use trigger info to build chat URL', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'custom-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          url: expect.stringContaining('/webhook/custom-chat'),
        })
      );
    });

    it('should use workflow ID as fallback when no trigger info', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
      };
      const workflow = createWorkflow();

      await handler.execute(input, workflow);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          url: expect.stringContaining('/webhook/workflow-123'),
        })
      );
    });

    it('should return error when base URL not available', async () => {
      const handlerNoContext = new ChatHandler(mockClient, {} as InstanceContext);

      // Mock getN8nApiConfig to return null
      const { getN8nApiConfig } = await import('../../../../src/config/n8n-api');
      vi.mocked(getN8nApiConfig).mockReturnValue(null as any);

      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();

      const response = await handlerNoContext.execute(input, workflow);

      expect(response.success).toBe(false);
      expect(response.error).toContain('Cannot determine n8n base URL');
    });

    it('should handle SSRF protection rejection', async () => {
      const { SSRFProtection } = await import('../../../../src/utils/ssrf-protection');
      vi.mocked(SSRFProtection.validateWebhookUrl).mockResolvedValue({
        valid: false,
        reason: 'Private IP address not allowed',
      });

      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();

      const response = await handler.execute(input, workflow);

      expect(response.success).toBe(false);
      expect(response.error).toContain('SSRF protection');
      expect(response.error).toContain('Private IP address not allowed');
    });

    it('should include additional data in payload', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        data: {
          userId: 'user-456',
          context: 'support',
        },
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          data: expect.objectContaining({
            action: 'sendMessage',
            chatInput: 'Hello',
            userId: 'user-456',
            context: 'support',
          }),
        })
      );
    });

    it('should pass custom headers', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        headers: {
          'X-Custom-Header': 'custom-value',
          'Authorization': 'Bearer token',
        },
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          headers: expect.objectContaining({
            'X-Custom-Header': 'custom-value',
            'Authorization': 'Bearer token',
            'Content-Type': 'application/json',
          }),
        })
      );
    });

    it('should use custom timeout when provided', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        timeout: 90000,
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          timeout: 90000,
        })
      );
    });

    it('should use default timeout of 120000ms when waiting for response', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        waitForResponse: true,
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          timeout: 120000,
        })
      );
    });

    it('should use timeout of 30000ms when not waiting for response', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
        waitForResponse: false,
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          timeout: 30000,
        })
      );
    });

    it('should return response with status and metadata', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello AI!',
        sessionId: 'session-123',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      vi.mocked(axios.request).mockResolvedValue({
        status: 200,
        statusText: 'OK',
        data: { response: 'AI reply', tokens: 150 },
      });

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(true);
      expect(response.status).toBe(200);
      expect(response.statusText).toBe('OK');
      expect(response.data).toEqual({ response: 'AI reply', tokens: 150 });
      expect(response.metadata?.duration).toBeGreaterThanOrEqual(0);
      expect(response.metadata?.sessionId).toBe('session-123');
      expect(response.metadata?.webhookPath).toBe('ai-chat');
    });

    it('should handle API errors gracefully', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      const apiError = new Error('Chat execution failed');
      vi.mocked(axios.request).mockRejectedValue(apiError);

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(false);
      expect(response.error).toBe('Chat execution failed');
    });

    it('should extract execution ID from error response', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      const apiError: any = new Error('Execution error');
      apiError.response = {
        data: {
          executionId: 'exec-789',
          error: 'Node failed',
        },
      };
      vi.mocked(axios.request).mockRejectedValue(apiError);

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(false);
      expect(response.executionId).toBe('exec-789');
      expect(response.details).toEqual({
        executionId: 'exec-789',
        error: 'Node failed',
      });
    });

    it('should handle error with code', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      const apiError: any = new Error('Timeout error');
      apiError.code = 'ETIMEDOUT';
      vi.mocked(axios.request).mockRejectedValue(apiError);

      const response = await handler.execute(input, workflow, triggerInfo);

      expect(response.success).toBe(false);
      expect(response.code).toBe('ETIMEDOUT');
    });

    it('should validate status codes less than 500', async () => {
      const input = {
        workflowId: 'workflow-123',
        triggerType: 'chat' as const,
        message: 'Hello',
      };
      const workflow = createWorkflow();
      const triggerInfo: DetectedTrigger = {
        type: 'chat',
        node: workflow.nodes[0],
        webhookPath: 'ai-chat',
      };

      await handler.execute(input, workflow, triggerInfo);

      expect(axios.request).toHaveBeenCalledWith(
        expect.objectContaining({
          validateStatus: expect.any(Function),
        })
      );

      const config = vi.mocked(axios.request).mock.calls[0][0];
      expect(config.validateStatus!(200)).toBe(true);
      expect(config.validateStatus!(404)).toBe(true);
      expect(config.validateStatus!(499)).toBe(true);
      expect(config.validateStatus!(500)).toBe(false);
      expect(config.validateStatus!(503)).toBe(false);
    });
  });
});
